import json
from pyecharts.charts import Line
from pyecharts.options import TitleOpts,LabelOpts,LegendOpts,ToolboxOpts,VisualMapOpts
# 处理数据
f_us = open("E://桌面/pythonProject/基础/数据可视化/数据/美国.txt", "r", encoding="UTF-8")
us_data = f_us.read()

f_in = open("E://桌面/pythonProject/基础/数据可视化/数据/印度.txt", "r", encoding="UTF-8")
in_data = f_in.read()

f_jp = open("E://桌面/pythonProject/基础/数据可视化/数据/日本.txt", "r", encoding="UTF-8")
jp_data = f_jp.read()

# 去除不规范开头
us_data = us_data.replace("jsonp_1629344292311_69436(","")
in_data = in_data.replace("jsonp_1629350745930_63180(","")
jp_data = jp_data.replace("jsonp_1629350871167_29498(","")

# 去除不规范结尾
us_data = us_data[:-2]
in_data = in_data[:-2]
jp_data = jp_data[:-2]

# json转换Python字符串
us_dict = json.loads(us_data)
in_dict = json.loads(in_data)
jp_dict = json.loads(jp_data)

# 获取trend key
us_trend_data = us_dict['data'][0]['trend']
in_trend_data = in_dict['data'][0]['trend']
jp_trend_data = jp_dict['data'][0]['trend']

# 获取日期数据，用于x轴，取2020年（到314结束）
us_x_data = us_trend_data['updateDate'][:314]
in_x_data = in_trend_data['updateDate'][:314]
jp_x_data = jp_trend_data['updateDate'][:314]

# 获取确诊数据，用于y轴，取2020年（到314结束）
us_y_data = us_trend_data['list'][0]['data'][:314]
in_y_data = in_trend_data['list'][0]['data'][:314]
jp_y_data = jp_trend_data['list'][0]['data'][:314]

# 生成图表
line = Line()       # 构建折线图对象
# 添加x轴数据，x轴是公用的，添加一份数据即可
line.add_xaxis(us_x_data)
# 添加y轴数据
line.add_yaxis("美国确诊人数",us_y_data,label_opts=LabelOpts(is_show=False))
line.add_yaxis("印度确诊人数",in_y_data,label_opts=LabelOpts(is_show=False))
line.add_yaxis("日本确诊人数",jp_y_data,label_opts=LabelOpts(is_show=False))

# 设置全局选项
line.set_global_opts(
    # 标题设置
    title_opts=TitleOpts(title="2020年美印日三国确诊人数对比",pos_left="center",pos_bottom="1%")
)

# 调用render方法生成图表
line.render()

# 关闭文件
f_us.close()
f_in.close()
f_jp.close()